# -*-Python-*-
# data-parallelism or mixture-of-experts on 128-core TPU

utils.run.mesh_shape = "all:128"
utils.run.layout_rules = "batch:all,experts:all"
